package com.qd.flow.service.config;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.db.BaseService;
import com.cdqidi.dict.OperatorDictTemplate;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.RedisTemplateUtil;
import com.qd.common.flow.domain.config.FlowProcessConfigDTO;
import com.qd.common.flow.domain.config.FlowProcessNodeDTO;
import com.qd.common.sys.enums.BaseTableEnum;
import com.qd.common.sys.util.SecurityUtils;
import com.qd.flow.mapper.config.FlowProcessConfigMapper;
import com.qd.flow.model.config.FlowProcessConfig;
import com.qd.flow.service.bill.FlowBillService;
import com.qd.system.service.dict.MbUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import javax.annotation.Resource;
import java.time.LocalDateTime;
import java.util.*;

import static java.util.stream.Collectors.toList;
import static java.util.stream.Collectors.toSet;

/**
 * 流程配置
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class FlowProcessConfigService extends BaseService<FlowProcessConfigMapper, FlowProcessConfig> {

    private static final String PREFIX_ID = "flow_process_config_processId_{0}";
    private static final String PREFIX_UNIQUE_FLOW_PR_CONFIG_PROCESS_CODE_I = "flow_process_config_processCode_{0}";

    private final RedisTemplateUtil<FlowProcessConfig> redisTemplateUtil;
    private final FlowProcessNodeService processNodeService;
    private final FlowBillService flowBillService;
    @Resource
    private OperatorDictTemplate operatorDictTemplate;

    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public FlowProcessConfigDTO save(FlowProcessConfigDTO dto) {
        validator(dto);
        checkUnique(dto);
        final FlowProcessConfig model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                saveNode(dto, model.getProcessId());
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }


    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public FlowProcessConfigDTO update(FlowProcessConfigDTO dto) {
        validator(dto);
        checkUnique(dto);
        final FlowProcessConfig historyModel = Optional.ofNullable(getById(dto.getProcessId())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final FlowProcessConfig model = dtoToModel(dto);
        try {
            if (super.updateById(model)) {
                saveNode(dto, model.getProcessId());
                return modelToDto(model);
            }
            throw new ApiException("更新失败");
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
            model.freeData();
        }
    }

    /**
     * 批量添加
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void saveBatch(List<FlowProcessConfig> mList) {
        try {
            if (!super.saveBatch(mList)) {
                throw new ApiException("批量添加失败");
            }
        } finally {
            clearCaches(mList);
        }
    }

    /**
     * 批量更新
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void updateBatchIds(List<FlowProcessConfig> mList) {
        final List<String> ids = mList.stream().map(FlowProcessConfig::getProcessId).collect(toList());
        final List<FlowProcessConfig> historyModelList = checkIds(ids);
        try {
            if (!super.updateBatchById(mList)) {
                throw new ApiException("批量更新失败");
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
            ids.clear();
        }
    }

    /**
     * 删除，单个
     *
     * @param processId 主键
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String processId) {
        if (Boolean.TRUE.equals(flowBillService.running(processId))) {
            throw new ApiException("存在正在进行中的流程，不能删除");
        }
        final FlowProcessConfig historyModel = Optional.ofNullable(getById(processId)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(processId)) {
                throw new ApiException("删除失败");
            }
            processNodeService.removeByProcessId(processId);
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
        }
    }

    /**
     * 批量删除
     *
     * @param ids 主键集合
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeBatchIds(List<String> ids) {
        ids.forEach(this::removeById);
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(FlowProcessConfigDTO dto) {
        final QueryWrapper<FlowProcessConfig> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param processId 主键
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String processId) {
        boolean exists = false;
        if (StringUtils.hasLength(processId)) {
            final FlowProcessConfig model = getById(processId);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param processId 主键
     * @return 单条记录
     */
    public FlowProcessConfig getById(String processId) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, processId);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(processId), FlowProcessConfig.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param processId 主键
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public FlowProcessConfigDTO getByIdDto(String processId) {
        return Optional.ofNullable(getById(processId)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    public FlowProcessConfig getOne(FlowProcessConfigDTO dto) {
        final QueryWrapper<FlowProcessConfig> queryWrapper = createQueryWrapper(dto);
        FlowProcessConfig model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public FlowProcessConfigDTO getOneDto(FlowProcessConfigDTO dto) {
        return Optional.ofNullable(getOne(dto)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据
     */
    public List<FlowProcessConfig> getList(FlowProcessConfigDTO dto) {
        final QueryWrapper<FlowProcessConfig> queryWrapper = createQueryWrapper(dto);
        final List<FlowProcessConfig> list = super.list(queryWrapper);
        queryWrapper.clear();
        return list;
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据，转换成页面对象
     */
    @Transactional(readOnly = true)
    public List<FlowProcessConfigDTO> getListDto(FlowProcessConfigDTO dto) {
        return modelToDtoList(getList(dto));
    }

    /**
     * 分页数据
     *
     * @param page 分页对象
     * @param dto  查询参数
     * @return 分页信息
     */
    @Transactional(readOnly = true)
    public IPage<FlowProcessConfigDTO> page(Page<FlowProcessConfig> page, FlowProcessConfigDTO dto) {
        final QueryWrapper<FlowProcessConfig> queryWrapper = createQueryWrapper(dto);
        Page<FlowProcessConfig> iPage = baseMapper.selectPage(page, queryWrapper);
        Optional.ofNullable(iPage.getRecords()).ifPresent(this::modelToDtoList);
        queryWrapper.clear();
        return iPage.convert(this::modelToDto);
    }

    /**
     * 根据唯一性索引获得数据
     * param processCode
     *
     * @return 单条数据
     */
    public FlowProcessConfig getUniqueByProcessCode(String processCode) {
        final QueryWrapper<FlowProcessConfig> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("process_code", processCode);
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_FLOW_PR_CONFIG_PROCESS_CODE_I, processCode);
        final FlowProcessConfig model = redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getOne(queryWrapper, false), FlowProcessConfig.class);
        queryWrapper.clear();
        return model;
    }

    /**
     * 验证唯一性索引
     *
     * @param dto 页面对象
     */
    public void checkUnique(FlowProcessConfigDTO dto) {
        FlowProcessConfig model = getUniqueByProcessCode(dto.getProcessCode());
        if (null != model) {
            checkUnique(dto, model);
        }
    }

    /**
     * 验证唯一性索引
     *
     * @param dto   页面对象
     * @param model 根据约束字段查询出来的数据库对象
     */
    private void checkUnique(FlowProcessConfigDTO dto, FlowProcessConfig model) {
        if (!StringUtils.hasLength(dto.getProcessId())) {
            throw new ApiException("数据已存在，请勿重复操作");
        }
        if (!dto.getProcessId().equals(model.getProcessId())) {
            throw new ApiException("数据已存在，请勿重复操作");
        }
    }

    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(FlowProcessConfig model) {
        if (null == model) {
            return;
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getProcessId()));
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_FLOW_PR_CONFIG_PROCESS_CODE_I, model.getProcessCode()));
    }

    /**
     * 批量清空缓存
     *
     * @param list 数据库对象
     */
    public void clearCaches(List<FlowProcessConfig> list) {
        if (null == list || list.isEmpty()) {
            return;
        }
        List<String> ids = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getProcessId())).collect(toList());
        if (!ids.isEmpty()) {
            redisTemplateUtil.deleteByKey(ids);
            ids.clear();
        }
        Set<String> uniqueIds = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_FLOW_PR_CONFIG_PROCESS_CODE_I, model.getProcessCode())).collect(toSet());
        if (!uniqueIds.isEmpty()) {
            redisTemplateUtil.deleteByKey(uniqueIds);
            uniqueIds.clear();
        }
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public FlowProcessConfigDTO modelToDto(FlowProcessConfig model) {
        if (null == model) {
            return null;
        }
        final FlowProcessConfigDTO dto = new FlowProcessConfigDTO();
        dto.setProcessId(model.getProcessId());
        dto.setProcessCode(model.getProcessCode());
        dto.setProcessName(model.getProcessName());
        dto.setAddTime(model.getAddTime());
        dto.setUpdateTime(model.getUpdateTime());
        dto.setOperator(model.getOperator());
        dto.setOrgId(model.getOrgId());
        dto.setEnable(model.getEnable());
        buildDisplay(dto);
        return dto;
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public FlowProcessConfig dtoToModel(FlowProcessConfigDTO dto) {
        if (null == dto) {
            return null;
        }
        final FlowProcessConfig model = new FlowProcessConfig();
        model.setProcessId(dto.getProcessId());
        model.setProcessCode(dto.getProcessCode());
        model.setProcessName(dto.getProcessName());
        model.setAddTime(dto.getAddTime());
        model.setUpdateTime(dto.getUpdateTime());
        model.setOperator(dto.getOperator());
        model.setOrgId(dto.getOrgId());
        model.setEnable(dto.getEnable());
        return model;
    }

    /**
     * 码表转换
     *
     * @param dto 页面对象
     */
    public void buildDisplay(FlowProcessConfigDTO dto) {
        operatorDictTemplate.execute(dto);
        List<FlowProcessNodeDTO> nodeList = processNodeService.getByProcessIdDTO(dto.getProcessId());
        if (!nodeList.isEmpty()) {
            dto.setNodeList(nodeList);
        }
    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<FlowProcessConfigDTO> modelToDtoList(List<FlowProcessConfig> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<FlowProcessConfig> dtoToModelList(List<FlowProcessConfigDTO> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<FlowProcessConfig> createQueryWrapper(FlowProcessConfigDTO dto) {
        final QueryWrapper<FlowProcessConfig> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(dto.getProcessCode())) {
                queryWrapper.eq("process_code", dto.getProcessCode());
            }
            if (StringUtils.hasLength(dto.getProcessName())) {
                queryWrapper.like("process_name", dto.getProcessName());
            }
            if (StringUtils.hasLength(dto.getEnable())) {
                queryWrapper.eq("enable", dto.getEnable());
            }
        }
        queryWrapper.orderByDesc("add_time");
        return queryWrapper;
    }

    /**
     * 批量检查主键在数据库中是否存在
     *
     * @param ids 主键集合
     * @return 主键查询出来的集合
     */
    public List<FlowProcessConfig> checkIds(List<String> ids) {
        if (null == ids || ids.isEmpty()) {
            throw new ApiException("ids不能为空");
        }
        final List<FlowProcessConfig> list = super.listByIds(ids);
        if (list.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(ids.toArray()));
        }
        if (ids.size() == list.size()) {
            return list;
        }
        List<String> dbIds = list.stream().map(FlowProcessConfig::getProcessId).collect(toList());
        final List<String> notExistsList = ids.stream().filter(id -> !dbIds.contains(id)).collect(toList());
        dbIds.clear();
        if (!notExistsList.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(notExistsList.toArray()));
        }
        return list;
    }


    private void validator(FlowProcessConfigDTO dto) {
        if (StringUtils.hasLength(dto.getEnable())) {
            if (MbUtil.isNotExists(dto.getEnable(), BaseTableEnum.SF)) {
                throw new ApiException("是否启用值错误");
            }
        } else {
            dto.setEnable(ConstantDto.SF1);
        }
        dto.setOperator(SecurityUtils.getLoginUser().getUserId());
        dto.setOrgId(SecurityUtils.getLoginUser().getCurrentOrgId());
    }

    private void saveNode(FlowProcessConfigDTO dto, String processId) {
        List<FlowProcessNodeDTO> nodeList = dto.getNodeList();
        if (null != nodeList && !nodeList.isEmpty()) {
            LocalDateTime now = LocalDateTime.now();
            nodeList.forEach(node -> {
                node.setProcessId(processId);
                node.setEnable(dto.getEnable());
                node.setOperator(dto.getOperator());
                node.setOrgId(dto.getOrgId());
                if (StringUtils.hasLength(node.getNodeId())) {
                    if (null == dto.getUpdateTime()) {
                        node.setUpdateTime(now);
                    } else {
                        node.setUpdateTime(dto.getUpdateTime());
                    }
                    processNodeService.update(node);
                } else {
                    if (null == dto.getAddTime()) {
                        node.setAddTime(now);
                    } else {
                        node.setAddTime(dto.getAddTime());
                    }
                    processNodeService.save(node);
                }
                node.freeData();
            });
            nodeList.clear();
        }
    }

    @Transactional(readOnly = true)
    public String idToDisplay(String processId) {
        return Optional.ofNullable(processId).map(this::getById).map(FlowProcessConfig::getProcessName).orElse("");
    }
}